package cn.itcast.bos.web.action.base;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;

import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Row;
import org.apache.struts2.convention.annotation.Action;
import org.apache.struts2.convention.annotation.Namespace;
import org.apache.struts2.convention.annotation.ParentPackage;
import org.apache.struts2.convention.annotation.Result;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Controller;

import cn.itcast.bos.domain.base.Area;
import cn.itcast.bos.service.base.AreaService;
import cn.itcast.bos.utils.PinYin4jUtils;
import cn.itcast.bos.web.action.common.BaseAction;
@ParentPackage("json-default")
@Namespace("/")
@Controller
@Scope("prototye")
public class AreaAction extends BaseAction<Area>{

	
	@Autowired
	private AreaService areaService;
	

	// 接收上传文件
	private File file;

	public void setFile(File file) {
		this.file = file;
	}

	//批量导入数据：
	@Action(value = "area_batchImport")
	public String batchImport() throws  IOException{
		List<Area> list = new ArrayList<Area>();
		//获取excle对象：
		HSSFWorkbook hssfWorkbook = new HSSFWorkbook(new FileInputStream(file));
		//获取sheet
		HSSFSheet sheet = hssfWorkbook.getSheetAt(0);
		//遍历得到每一行
		for (Row row : sheet) {
			if(row.getRowNum() ==0){
				//跳过第一行
				continue;
			}
			//跳过空行
			if(row.getCell(0)==null||StringUtils.isBlank(row.getCell(0).getStringCellValue())){
				continue;
			}
			//获取每个单元格的数据：
			Area area = new Area();
			area.setId(row.getCell(0).getStringCellValue());
			area.setProvince(row.getCell(1).getStringCellValue());
			area.setCity(row.getCell(2).getStringCellValue());
			area.setDistrict(row.getCell(3).getStringCellValue());
			area.setPostcode(row.getCell(4).getStringCellValue());
			
			//基于pinyin4j生成简码
			String province = area.getProvince();
			String city = area.getCity();
			String district = area.getDistrict();
			province = province.substring(0, province.length()-1);
			city = city.substring(0, city.length()-1);
			district = district.substring(0, district.length()-1);
			
			//简码
			String[] headArray = PinYin4jUtils.getHeadByString(province + city +district);
			StringBuffer stringBuffer = new StringBuffer();
			for (String str : headArray) {
				stringBuffer.append(str);
			}
			
			String shortcode = stringBuffer.toString();
			area.setShortcode(shortcode);
			//生成编码；
			String citycode = PinYin4jUtils.hanziToPinyin(city,"");
			area.setCitycode(citycode);
			list.add(area);
		}
		
		// 调用业务层
				areaService.saveBatch(list);
		return NONE;
		
	}
	
	@Action(value = "area_pageQuery",results = {@Result(name = "success",type = "json")})
	public String pageQuery(){
		
		Specification<Area> specification = new Specification<Area>() {
			
			@Override
			public Predicate toPredicate(Root<Area> root, CriteriaQuery<?> query,
					CriteriaBuilder cb) {
				
				List <Predicate> list  = new ArrayList<Predicate>();
				if(StringUtils.isNotBlank(model.getProvince())){
					Predicate p1 = cb.equal(root.get("province").as(String.class), model.getProvince());
					list.add(p1);
				};
				
				if(StringUtils.isNotBlank(model.getCity())){
					Predicate p2 = cb.equal(root.get("city").as(String.class), model.getCity());
					list.add(p2);
				}
				
				if(StringUtils.isNotBlank(model.getDistrict())){
					Predicate p3 = cb.like(root.get("district").as(String.class),"%"+model.getDistrict()+"%");
					list.add(p3);
				}
				
				return cb.and(list.toArray(new Predicate[0]));
			}
		};
		
		Pageable pageable = new PageRequest(page-1, rows);
		Page<Area> pageData = areaService.findPageData(specification,pageable);
		
		pushPageDataToValueStack(pageData);
		return SUCCESS;
		
	}

}
